package main

import (
        "fmt"
        "os"
        "os/exec"
)

func main() {
        // 指定要生成代码的路径
        schemaPath := "./ent/schema"

        // 指定生成的目标目录
        targetDir := "./ent"

        // 构造命令
        cmd := exec.Command("ent", "generate", schemaPath, "--target", targetDir)

        // 设置环境变量（如果需要）
        // os.Setenv("ENT_FEATURE", "feature1,feature2")

        // 执行命令
        output, err := cmd.CombinedOutput()
        if err != nil {
                fmt.Printf("Error running ent generate: %s\n", err)
                fmt.Printf("Output: %s\n", output)
                os.Exit(1)
        }

        // 输出结果
        fmt.Printf("Ent code generated successfully in %s\n", targetDir)
        fmt.Printf("Output: %s\n", output)
}